package cn.itsource.util;

import cn.itsource.domain.User;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 创建一个用户上下文类，专门用来存储登录用户对象和获取登录对象
 */
public class UserContext {

    //创建一个map集合
    private static Map<String, User> concurrentHashMap = new ConcurrentHashMap<>();

    /**
     * 把登录对象放到map集合中
     * @param token
     * @param user
     */
    public static void setLoginUser(String token,User user){
        concurrentHashMap.put(token, user);
    }

    /**
     * 根据token获取登录对象
     * @param token
     * @return
     */
    public static User getLoginUser(String token) {
        return concurrentHashMap.get(token);
    }
}
